/*
 * Copyright (c) Huawei Technologies Co., Ltd. 2023-2023. All rights reserved.
 * Description: 与AI设备(D芯片)相关hdds接口适配INC头文件
 * Author: linzhiyang
 * Create: 2023-01-17
 */

#ifndef HCCL_INC_ADAPTER_HDDS_H
#define HCCL_INC_ADAPTER_HDDS_H

#include <hccl/hccl_types.h>

#include "hccl/base.h"

HcclResult hrtQueryPartitionMapPsId(u64 key, u32 *psId);
HcclResult hrtInitPartitionMap(u32 partitionNum, u32 psNum, const u32 psId[]);
HcclResult hrtGetBatchPsIds(u64 *keys, u32 *psIds[], u32 num);
#endif
